package com.chrray.position.util.excel;

import cn.afterturn.easypoi.excel.ExcelImportUtil;
import cn.afterturn.easypoi.excel.entity.ImportParams;
import com.chrray.position.util.result.ResponseResult;
import com.chrray.position.util.result.ResponseResults;

import java.io.InputStream;

/**
 * @author lm
 * @anthor ch
 * @description 导入excel
 * @date
 */
public class ImportExcelUtils<T> {

    /**
     * excel导入，不设置参数，使用默认参数
     *
     * @param pojoClass   pojo类型
     * @param inputstream 文件流
     *                    对应springboot的MultipartFile file
     *                    file.getInputStream()
     */
    public static ResponseResult importExcel(InputStream inputstream, Class<?> pojoClass) {
        ImportParams params = new ImportParams();
        params.setHeadRows(1);
        params.setTitleRows(0);
        try {
            ExcelImportUtil.importExcel(inputstream, pojoClass, params);
            return ResponseResults.success("导入数据成功");
        } catch (Exception e) {
            return ResponseResults.fail("导入数据失败");
        }
    }

    /**
     * excel导入，自定义参数，使用默认参数
     *
     * @param pojoClass   pojo类型
     * @param inputstream 文件流
     *                    对应springboot的MultipartFile file
     *                    file.getInputStream()
     */
    public static ResponseResult importExcel(InputStream inputstream, Class<?> pojoClass, ImportParams params) {
        try {
            ExcelImportUtil.importExcel(inputstream, pojoClass, params);
            return ResponseResults.success("导入数据成功");
        } catch (Exception e) {
            return ResponseResults.fail("导入数据失败");
        }
    }

}
